package 算法基础模板.chapter_01.双指针.sliding_window;

/**
 * 992. K 个不同整数的子数组
 * 给定一个正整数数组 A，如果 A 的某个子数组中不同整数的个数恰好为 K，则称 A 的这个连续、不一定独立的子数组为好子数组。
 * <p>
 * （例如，[1,2,3,1,2] 中有 3 个不同的整数：1，2，以及 3。）
 * <p>
 * 返回 A 中好子数组的数目。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：A = [1,2,1,2,3], K = 2
 * 输出：7
 * 解释：恰好由 2 个不同整数组成的子数组：[1,2], [2,1], [1,2], [2,3], [1,2,1], [2,1,2], [1,2,1,2].
 * 示例 2：
 * <p>
 * 输入：A = [1,2,1,3,4], K = 3
 * 输出：3
 * 解释：恰好由 3 个不同整数组成的子数组：[1,2,1,3], [2,1,3], [1,3,4].
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= A.length <= 20000
 * 1 <= A[i] <= A.length
 * 1 <= K <= A.length
 *
 * @author Summerday
 */
public class LC992 {
    public int subarraysWithKDistinct (int[] A, int K) {
        return atMostK(A, K) - atMostK(A, K - 1);
    }

    int atMostK (int[] A, int K) {
        int res = 0;
        int[] count = new int[A.length + 1];
        for (int i = 0, j = 0; j < A.length; ++j) {
            if (count[A[j]] == 0) K--; // 说明j这个位置的数使第一次出现
            count[A[j]]++;
            while (K < 0) { // 表示数的出现次数出现太多，左指针右移
                count[A[i]]--;
                if (count[A[i]] == 0) { // 表示左指针这个只有一类，k恢复一下
                    K++;
                }
                i++;
            }
            res += j - i + 1;
        }
        return res;
    }
}
